package org.jeecg.modules.system.rule;

import com.alibaba.fastjson.JSONObject;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import io.netty.util.internal.StringUtil;
import org.jeecg.common.handler.IFillRuleHandler;
import org.jeecg.common.util.SpringContextUtils;
import org.jeecg.common.util.YouBianCodeUtil;
import org.jeecg.modules.zcgl.entity.Region;
import org.jeecg.modules.zcgl.service.IRegionService;

import java.util.ArrayList;
import java.util.List;

/**
 * @Author scott
 * @Date 2019/12/9 11:33
 * @Description: 区域编码生成规则
 */
public class RegionCodeRule implements IFillRuleHandler {

    @Override
    public Object execute(JSONObject params, JSONObject formData) {

        IRegionService regionServiceImpl = (IRegionService) SpringContextUtils.getBean("regionServiceImpl");

        LambdaQueryWrapper<Region> query = new LambdaQueryWrapper<>();
        LambdaQueryWrapper<Region> query1 = new LambdaQueryWrapper<>();
        // 创建一个List集合,存储查询返回的所有SysDepart对象
        List<Region> regionList = new ArrayList<>();
        String strArray = "";
        // 定义新编码字符串
        String newRegionCode = "";
        // 定义旧编码字符串
        String oldRegionCode = "";

        String parentId = null;
        if (formData != null && formData.size() > 0) {
            Object obj = formData.get("parentId");
            if (obj != null) parentId = obj.toString();
        } else {
            if (params != null) {
                Object obj = params.get("parentId");
                if (obj != null) parentId = obj.toString();
            }
        }

        //如果是最高级,则查询出同级的org_code, 调用工具类生成编码并返回
        if (StringUtil.isNullOrEmpty(parentId)) {
            // 线判断数据库中的表是否为空,空则直接返回初始编码
            query1.eq(Region::getParentId, "").or().isNull(Region::getParentId);
            query1.orderByDesc(Region::getRegionCode);
            regionList = regionServiceImpl.list(query1);
            if (regionList == null || regionList.size() == 0) {
                strArray = YouBianCodeUtil.getNextYouBianCode(null);
                return strArray;
            } else {
                Region depart = regionList.get(0);
                oldRegionCode = depart.getRegionCode();
                newRegionCode = YouBianCodeUtil.getNextYouBianCode(oldRegionCode);
            }
        } else {//反之则查询出所有同级的区域,获取结果后有两种情况,有同级和没有同级
            // 封装查询同级的条件
            query.eq(Region::getParentId, parentId);
            // 降序排序
            query.orderByDesc(Region::getRegionCode);
            // 查询出同级区域的集合
            List<Region> parentList = regionServiceImpl.list(query);
            // 查询出父级区域
            Region region = regionServiceImpl.getById(parentId);
            // 获取父级区域的Code
            String parentCode = region.getRegionCode();
            // 根据父级区域类型算出当前区域的类型
            // 处理同级区域为null的情况
            if (parentList == null || parentList.size() == 0) {
                // 直接生成当前的区域编码并返回
                newRegionCode = YouBianCodeUtil.getSubYouBianCode(parentCode, null);
            } else { //处理有同级区域的情况
                // 获取同级区域的编码,利用工具类
                String subCode = parentList.get(0).getRegionCode();
                // 返回生成的当前区域编码
                newRegionCode = YouBianCodeUtil.getSubYouBianCode(parentCode, subCode);
            }
        }
        // 返回最终封装了区域编码和区域类型的数组
        strArray = newRegionCode;
        return strArray;
    }
}
